Skip to content

docs: qualify SALM Automodel parallelism support - #15976

Merged
blisc merged 1 commit into
mainfrom
codex/qualify-salm-parallelism-docs
Jul 28, 2026
Merged

docs: qualify SALM Automodel parallelism support#15976
blisc merged 1 commit into
mainfrom
codex/qualify-salm-parallelism-docs

Conversation

@DongjiGao

Copy link
Copy Markdown
Contributor

What changed

  • Clarify that AutomodelParallelStrategy can construct meshes for FSDP2, TP, PP, CP, EP, and HSDP, while actual support remains model-, backend-, and combination-dependent.
  • Document the validated native Nemotron-V3 SALMAutomodel envelope: FSDP2, EP, and CP with packed sequences and Transformer Engine attention.
  • Direct users to keep tp_size, dp_replicate_size, and the currently unvalidated pp_size at 1 for this configuration.

Why

The previous blanket support statement drove QA and users toward unsupported Nemotron-V3 configurations. Nemotron-V3 has no TP plan, while the SALMAutomodel HSDP path is unsupported and currently fails with an opaque DeviceMesh error. PP is capability-signaled by Automodel but has no SpeechLM2 training validation.

This is a documentation correction only; it does not change parallelism behavior. The HSDP validation/runtime issue remains separate.

Validation

  • sphinx-lint docs/source/speechlm2/training_and_scaling.rst — no problems found
  • git diff --check

Related: internal bug 6437904.

Signed-off-by: Dongji Gao <dongjig@nvidia.com>
@copy-pr-bot

copy-pr-bot Bot commented Jul 28, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@DongjiGao
DongjiGao requested review from KunalDhawan, blisc and pzelasko and removed request for pzelasko July 28, 2026 18:51
@DongjiGao
DongjiGao marked this pull request as ready for review July 28, 2026 18:55

@blisc blisc left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Makes sense to me

@blisc blisc added the r3.0.0 Auto-cherrypick to release branch. Apply before merge; cherrypick happens after merge. label Jul 28, 2026
@blisc

blisc commented Jul 28, 2026

Copy link
Copy Markdown
Collaborator

Do you want to merge or keep as draft?

@DongjiGao

Copy link
Copy Markdown
Contributor Author

Do you want to merge or keep as draft?

I think this one is good to merge.

@blisc
blisc merged commit 9ae3e66 into main Jul 28, 2026
57 checks passed
@blisc
blisc deleted the codex/qualify-salm-parallelism-docs branch July 28, 2026 19:53
blisc pushed a commit that referenced this pull request Jul 28, 2026
Signed-off-by: Dongji Gao <dongjig@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: Dongji Gao <dongjig@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

r3.0.0 Auto-cherrypick to release branch. Apply before merge; cherrypick happens after merge.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants